
from django.shortcuts import render
from django.shortcuts import redirect
from django.http import HttpResponse
from bson import json_util
from django.contrib.auth.decorators import login_required
from django.contrib.auth.decorators import permission_required
from bson.objectid import ObjectId
import datetime

# import matplotlib.pyplot as plt
# plt.rcParams['font.sans-serif']=['SimHei']#设置字体以便支持中文
# import numpy as np
import os
from collections import OrderedDict

import json

import pymongo

#建立MongoDB数据库连接
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["test3"]

#各省首页
def pending_index(request,province):
    return render(request,"pending/"+province+"/index.html")

#根据前端url传来的数据，判断当前问卷类别，返回相应草稿箱数据库
def getMycol(no):
    if no == 'no1':
        return mydb["questionnaire11"] #草稿箱中的脑出血患者调查问卷存储
    elif no == 'no2':
        return mydb["questionnaire12"] #草稿箱中的健康人群调查问卷存储
    elif no == 'no4':
        return mydb["questionnaire14"] #草稿箱中的健康人群调查问卷(体检中心)存储
    elif no == 'no5':
        return mydb["questionnaire15"] #草稿箱中的脑出血患者调查问卷(医院)存储
    elif no == 'no6':
        return mydb["questionnaire16"] #草稿箱中的随访问卷存储
    elif no == 'no7':
        return mydb["questionnaire17"] #草稿箱中的入户筛查表
    elif no == 'no8':
        return mydb["questionnaire18"]  # 存储健康人群问卷（湖南省）
    elif no == 'no9':
        return mydb["questionnaire19"]  # 草稿箱中的健脑出血患者调查问卷（湖南省）
    elif no == 'no10':
        return mydb["questionnaire110"] # 草稿箱中的脑梗塞患者调查问卷（湖南省）
    else:
        return mydb["questionnaire111"]  # 草稿箱中的街道乡镇级调查单位基本信息采集表
#判断省各省名称
def getProvince(province):
    if province == 'hunan':
        return '湖南省'
    elif province == 'hubei':
        return '湖北省'
    elif province == 'henan':
        return '河南省'
    elif province == 'anhui':
        return '安徽省'
    else:
        return '江西省'

def pending(request,province,no):
    #根据前端url传来的数据，得到数据库和省的名称
    mycol1 = getMycol(no)
    province1 = getProvince(province)

    user_obj1 = mycol1.find({"所在的省": province1, '是否需要再次编辑': None})
    user_obj2 = mycol1.find({"所在的省": province1, '是否需要再次编辑': '是'})

    #存储待提交问卷，返回至前端页面
    user_obj = []
    for i in user_obj1:
        user_obj.append(i)
    for i in user_obj2:
        user_obj.append(i)

    return render(request, 'pending/'+province+'/'+no+'/index.html', {'user_obj': user_obj})

#由于问卷种类不同，采用专门函数处理湖南省数据
def pending_hunan(request,no):
    #根据前端url传来的数据，得到数据
    mycol1 = getMycol(no)

    user_obj1 = mycol1.find({'是否需要再次编辑': None})
    user_obj2 = mycol1.find({'是否需要再次编辑': '是'})

    #存储待提交问卷，返回至前端页面
    user_obj = []
    for i in user_obj1:
        user_obj.append(i)
    for i in user_obj2:
        user_obj.append(i)

    return render(request, 'pending/hunan/'+no+'/index.html', {'user_obj': user_obj})